#!/bin/bash

trtexec --onnx=./model.onnx \
        --saveEngine=./swin.engine \
        --minShapes=x:1x3x224x224 \
        --optShapes=x:16x3x224x224 \
        --maxShapes=x:16x3x224x224 \
        --workspace=1024
